package 简单.其他;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

/**
 * 给你一个整数数组 nums 和一个整数 k ，请你返回数对 (i, j) 的数目，满足 i < j 且 |nums[i] - nums[j]| == k 。
 * <p>
 * |x| 的值定义为：
 * <p>
 * 如果 x >= 0 ，那么值为 x 。
 * 如果 x < 0 ，那么值为 -x 。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/count-number-of-pairs-with-absolute-difference-k
 */
public class 差的绝对值为K的数对数目_2006 {

    public static void main(String[] args) {

        int[] nums = {1, 2, 2, 1};
//        System.out.println(new 差的绝对值为K的数对数目_2006().countKDifference(nums, 1));
        System.out.println(new 差的绝对值为K的数对数目_2006().countKDifference4(nums, 1));

    }

    public int countKDifference(int[] nums, int k) {
        int ans = 0;
        for (int i = 0; i < nums.length; i++) {
            int curNum = nums[i];
            for (int j = i + 1; j < nums.length; j++) {
                if (Math.abs(curNum - nums[j]) == k) {
                    ans++;
                }
            }
        }
        return ans;
    }

    /**
     * 先排序
     * 通过二分查找，找到等于nums[i]+k的的值中，索引的最小一个
     *
     * @param nums
     * @param k
     * @return
     */
    public int countKDifference2(int[] nums, int k) {
        int ans = 0;
        Arrays.sort(nums);
        for (int i = 0; i < nums.length; i++) {
            for (int j = search(nums, i, nums[i] + k); j < nums.length; j++) {
                if (nums[j] == nums[i] + k) {
                    ans++;
                } else {
                    break;
                }
            }
        }
        return ans;
    }

    public int search(int[] nums, int l, int destNum) {
        int ans = Integer.MAX_VALUE;
        int left = l + 1;
        int right = nums.length - 1;
        while (left <= right) {
            int mid = left + (right - left) / 2;
            if (nums[mid] == destNum) {
                ans = mid;  //找到可以满足条件的值
                right = mid - 1;
            } else if (nums[mid] < destNum) {
                left = mid + 1;
            } else {
                right = mid - 1;
            }
        }
        return ans;
    }

    /**
     * 用map存储某个值出现的次数
     * 只需要两次遍历
     *
     * @param nums
     * @param k
     * @return
     */
    public int countKDifference3(int[] nums, int k) {
        int ans = 0;
        Map<Integer, Integer> map = new HashMap<>();
        for (int num : nums) {
            map.put(num, map.getOrDefault(num, 0) + 1);
        }
        for (int num : nums) {
            ans += map.getOrDefault(num + k, 0);
        }
        return ans;
    }

    /**
     * 一次遍历
     * 用map存储之前，出现过的数字次数
     *
     * @param nums
     * @param k
     * @return
     */
    public int countKDifference4(int[] nums, int k) {
        int ans = 0;
        Map<Integer, Integer> map = new HashMap<>();
        for (int num : nums) {
            ans += map.getOrDefault(num - k, 0) + map.getOrDefault(num + k, 0);
            map.put(num, map.getOrDefault(num, 0) + 1);
        }
        return ans;
    }

}
